package org.pombo.gtb;

import java.util.List;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.pombo.gtm.Action;
import org.pombo.gtm.InformationSet;
import org.pombo.gtm.StrategyNode;

public class InformationSetBuilder extends Builder<InformationSet> {

	@Override
	public void clear() {
		this.entity = mock(InformationSet.class);
	}

	public InformationSetBuilder withNodes(List<StrategyNode> nodes) {
		when(this.entity.getNodes()).thenReturn(nodes);
		return this;
	}

	public InformationSetBuilder withActions(List<Action> actions) {
		when(this.entity.getActions()).thenReturn(actions);
		return this;
	}

}
